write a function in python hackerrank

welcome to shbcf.ru